package leet_code;

import java.util.HashMap;
import java.util.Map;

/**
 * 290. 单词规律
 */

public class WordPattern {
    // 给定一种规律 pattern 和一个字符串 str ，判断 str 是否遵循相同的规律。
    //这里的 遵循 指完全匹配，例如， pattern 里的每个字母和字符串 str 中的每个非空单词之间存在着双向连接的对应规律。

    public static void main(String[] args) {
        String pattern = "abba";
        String str1 = "dog dog dog dog";
//        String str2 = "cat dog cat dog";
        System.out.println(wordPattern(pattern, str1));
//        System.out.println(wordPattern(pattern, str2));
    }

    public static boolean wordPattern(String pattern, String str) {
        String[] strArr = str.split(" ");
        char[] patternArr = pattern.toCharArray();
        if (strArr.length != patternArr.length) {
            return false;
        }
        Map<Character, String> map = new HashMap<>();
        for (int i = 0; i < patternArr.length; i++) {
            if (!map.containsKey(patternArr[i]) && map.containsValue(strArr[i])) {
                return false;
            }
            if (!map.containsKey(patternArr[i])) {
                map.put(patternArr[i], strArr[i]);
            }
        }
        for (int i = 0; i < patternArr.length; i++) {
            if (!strArr[i].equals(map.get(patternArr[i]))) {
                return false;
            }
        }
        return true;
    }
}
